#定义全局变量
money = 5000000
name =None
#要求用户输入姓名:
name = input("请输入您的姓名:")
#查询余额函数
def query(show_header):
    if show_header:
        print("----------查询余额------------")
    print(f"{name},您好,您的余额剩余: {money}")
#存款函数
def set_money(num):
    global money
    money +=num
    print("----------存款------------")
    print(f"{name}, 您好, 您存款{num}成功")
    query(False)
#取款函数
def get_money(num):
    global money
    money -= num
    print("----------取款------------")
    print(f"{name}, 您好, 您取款{num}成功")
    query(False)
#主菜单函数
def main():
    print("----------主菜单------------")
    print(f"{name},您好,欢迎来到黑马银行ATM,请选择操作: ")
    print("查询余额  [输入1]")
    print("存款\t\t [输入2]")
    print("取款\t\t [输入3]")
    print("退出\t\t [输入4]")
    return input("请输入你的选择: ")

while(True):
    keyboard_input = main()
    if keyboard_input == "1":
        query(True)
        continue
    elif keyboard_input == "2":
        num = int(input("您想要存多少钱?请输入: "))
        set_money(num)
        continue
    elif keyboard_input == "3":
        num = int(input("您想要取多少钱?请输入: "))
        get_money(num)
        continue
    else:
        break


